package top.zhangjianyong.utils.file;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * @program: topzhangjianyongutils
 * @description: 文件下载
 * @author: 张建勇
 * @create: 2019-03-15 15:35
 **/
public class DownFileUtil {

    private URL fileUrl;// 文件下载路径
    private int threadCount;// 文件下载的线程数
    private int startPos;// 每个线程下载文件的开始位置
    private int size;// 每个线程下载文件的长度
    private int fileLength;// 文件总程度
    private String pathName;// 下载的文件路径（包含文件名）
    private Downthread[] tDownthreads;// 线程数组

    public DownFileUtil(URL url, int threadCount, String pathName) throws IOException {
        fileUrl = url;
        this.threadCount = threadCount;
        this.pathName = pathName;
        init();

    }

    /**
     * 初始化
     *
     * @throws IOException
     */
    private void init() throws IOException {
        tDownthreads = new Downthread[threadCount];
        HttpURLConnection conn = (HttpURLConnection) fileUrl.openConnection();
        conn.setConnectTimeout(5000);
        conn.setRequestMethod("GET");
        conn.setRequestProperty("connection", "keep-alive");
        fileLength = conn.getContentLength();
        System.out.println("文件长度" + fileLength);
        size = fileLength / threadCount;
        System.out.println("每个下载量==" + size);
        conn.disconnect();// 断开链接

    }

    public URL getFileUrl() {
        return fileUrl;
    }

    public int getThreadCount() {
        return this.threadCount;
    }

    /**
     * 开始下载
     */
    public void startDown() {
        for (int i = 0; i < threadCount; i++) {
            try {
                RandomAccessFile raFile = new RandomAccessFile(pathName, "rw");
                tDownthreads[i] = new Downthread(i * size, raFile, i);
                tDownthreads[i].start();
            } catch (FileNotFoundException e) {
                e.printStackTrace();
            }

        }

    }

    /**
     * 下载线程类
     *
     * @author luweicheng
     */
    class Downthread extends Thread {
        private int startPos;// 开始的位置
        private InputStream is;
        private RandomAccessFile raFile;
        private int length;// 下载的文件长度
        private int flag;// 线程标志

        public Downthread(int startPos, RandomAccessFile raFile, int i) {
            this.startPos = startPos;
            this.raFile = raFile;
            flag = i;
        }

        @Override
        public void run() {
            try {
                HttpURLConnection connection = (HttpURLConnection) fileUrl.openConnection();

                connection.setRequestMethod("GET");
                connection.setRequestProperty("connection", "keep-alive");
                connection.setConnectTimeout(5 * 1000);
                //设置请求数据的范围
                connection.setRequestProperty("Range", "bytes=" + startPos + "-" + (startPos + size));
                is = connection.getInputStream();
                //is.skip(startPos);
                raFile.seek(startPos);
                byte[] buf = new byte[8 * 1024];
                int hasread = 0;// 读出的字节数
                // 将位置在 startPos - startPos 位置的数据读出写入
                while (length < size && (hasread = is.read(buf)) != -1) {
                    raFile.write(buf, 0, hasread);
                    length += hasread;
                    System.out.println("*****线程" + flag + "下载了*********" + length);
                }
                System.out.println("*******线程" + flag + "下载完成*********");

            } catch (IOException e) {
                e.printStackTrace();
            } finally {

                try {
                    is.close();
                    raFile.close();
                } catch (IOException e) {

                    e.printStackTrace();
                }
            }

        }
    }


    public static void main(String[] args) {
        String urlStr = "http://down.winbaicai.com:88/DaBaiCai_v6.0_1812.zip";
        String pathName = "D:\\test\\123" + FileUtil.getExtensionByFileName(urlStr);
        try {
            URL url = new URL(urlStr);
            DownFileUtil downFileUtil = new DownFileUtil(url, 10, pathName);
            downFileUtil.init();
            downFileUtil.startDown();
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

}
